SAE on toy model no resampling-untied weights